// Copyright 2014 The Chromium Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

// A very simple packet builder class for building RTCP packets.
// Used for testing only.
#ifndef MEDIA_CAST_NET_RTCP_TEST_RTCP_PACKET_BUILDER_H_
#define MEDIA_CAST_NET_RTCP_TEST_RTCP_PACKET_BUILDER_H_

#include <stdint.h>
#include <vector>

#include "base/big_endian.h"
#include "base/macros.h"
#include "media/cast/net/cast_transport_defines.h"
#include "media/cast/net/rtcp/rtcp_defines.h"

namespace media {
namespace cast {

    // These values are arbitrary only for the purpose of testing.

    namespace {
        // Sender report.
        static const uint32_t kNtpHigh = 0x01020304;
        static const uint32_t kNtpLow = 0x05060708;
        static const uint32_t kRtpTimestamp = 0x10203040;
        static const uint32_t kSendPacketCount = 987;
        static const uint32_t kSendOctetCount = 87654;

        // Report block.
        static const int kLoss = 0x01000123;
        static const int kExtendedMax = 0x15678;
        static const int kTestJitter = 0x10203;
        static const uint32_t kLastSr = 0x34561234;
        static const uint32_t kDelayLastSr = 1000;

        // DLRR block.
        static const int kLastRr = 0x34561234;
        static const int kDelayLastRr = 1000;

        // NACK.
        static const int kMissingPacket = 34567;

        // CAST.
        static const uint32_t kAckFrameId = 17;
        static const uint32_t kLostFrameId = 18;
        static const uint32_t kFrameIdWithLostPackets = 19;
        static const int kLostPacketId1 = 3;
        static const int kLostPacketId2 = 5;
        static const int kLostPacketId3 = 12;
        static const uint8_t kFeedbackSeq = 1;
    } // namespace

    class TestRtcpPacketBuilder {
    public:
        TestRtcpPacketBuilder();

        void AddSr(uint32_t remote_ssrc, int number_of_report_blocks);
        void AddSrWithNtp(uint32_t remote_ssrc,
            uint32_t ntp_high,
            uint32_t ntp_low,
            uint32_t rtp_timestamp);
        void AddRr(uint32_t remote_ssrc, int number_of_report_blocks);
        void AddRb(uint32_t rtp_ssrc);

        void AddXrHeader(uint32_t remote_ssrc);
        void AddXrDlrrBlock(uint32_t remote_ssrc);
        void AddXrExtendedDlrrBlock(uint32_t remote_ssrc);
        void AddXrRrtrBlock();
        void AddXrUnknownBlock();
        void AddUnknownBlock();

        void AddNack(uint32_t remote_ssrc, uint32_t local_ssrc);
        void AddSendReportRequest(uint32_t remote_ssrc, uint32_t local_ssrc);

        void AddCast(uint32_t remote_ssrc,
            uint32_t local_ssrc,
            base::TimeDelta target_delay);
        void AddCst2(const std::vector<FrameId>& later_received_frames);
        void AddErrorCst2(); // With wrong identifier.
        void AddPli(uint32_t remote_ssrc, uint32_t local_ssrc);

        void AddReceiverLog(uint32_t remote_ssrc);
        void AddReceiverFrameLog(uint32_t rtp_timestamp,
            int num_events,
            uint32_t event_timesamp_base);
        void AddReceiverEventLog(uint16_t event_data,
            CastLoggingEvent event,
            uint16_t event_timesamp_delta);

        std::unique_ptr<Packet> GetPacket();
        const uint8_t* Data();
        int Length() { return kMaxIpPacketSize - big_endian_writer_.remaining(); }
        base::BigEndianReader* Reader();

    private:
        void AddRtcpHeader(int payload, int format_or_count);
        void PatchLengthField();

        // Where the length field of the current packet is.
        // Note: 0 is not a legal value, it is used for "uninitialized".
        uint8_t buffer_[kMaxIpPacketSize];
        char* ptr_of_length_;
        base::BigEndianWriter big_endian_writer_;
        base::BigEndianReader big_endian_reader_;

        DISALLOW_COPY_AND_ASSIGN(TestRtcpPacketBuilder);
    };

} // namespace cast
} // namespace media

#endif // MEDIA_CAST_NET_RTCP_TEST_RTCP_PACKET_BUILDER_H_
